use scraper::{ Html, Selector };
use serde::{ Deserialize, Serialize };

use crate::REQWEST_CLIENT;
use anyhow::{ anyhow, Result };
/// 利润表
/// Mapping to [FUND_lrfpb_000001.js](file:///Users/CC/gitee/conanchen/fuquants-workspace/crates/fudacli/eastmoney/sample/FUND_lrfpb_000001.js)
#[derive(Debug, Serialize, Deserialize)]
struct ApiData {
    content: String,
    arryear: Vec<u32>,
    curyear: Option<u32>,
}

#[derive(Debug, Serialize, Deserialize, Default)]
pub struct FundLrfpb {
    pub report_date: String,
    pub data: Vec<FundLrfpbData>,
}

#[derive(Debug, Serialize, Deserialize, Default, Clone)]
pub struct FundLrfpbData {
    pub item_title: String,
    pub value: String,
}

impl FundLrfpb {
    /// Referer: https://fundf10.eastmoney.com/lrfpb_000001.html 利润表
    ///
    /// Data: https://fundf10.eastmoney.com/FundArchivesDatas.aspx?type=lrfpb&code=000001&showtype=1&year=2021&rt=0.262362624583663
    pub async fn fund_lrfpbs(code: &str, year: &str) -> Result<Vec<FundLrfpb>> {
        // GET /FundArchivesDatas.aspx?type=lrfpb&code=000001&showtype=1&year=2021&rt=0.262362624583663 HTTP/1.1
        // Accept: */*
        // Accept-Encoding: gzip, deflate, br
        // Accept-Language: zh-CN,zh;q=0.9,en;q=0.8
        // Cache-Control: no-cache
        // Connection: keep-alive
        // Cookie: qgqp_b_id=d6299473145338f68687834ba37fa719; st_si=77687174285423; ASP.NET_SessionId=5ufgex2sjbo4p44lwytenq3s; searchbar_code=159834; AUTH_FUND.EASTMONEY.COM_GSJZ=AUTH*TTJJ*TOKEN; EMFUND0=01-29%2015%3A02%3A07@%23%24%u534E%u590F%u4E2D%u8BC1%u5168%u6307%u623F%u5730%u4EA7ETF@%23%24515060; EMFUND1=01-30%2014%3A41%3A22@%23%24%u534E%u6CF0%u67CF%u745E%u6D88%u8D39%u6210%u957F%u6DF7%u5408@%23%24001069; EMFUND2=01-30%2014%3A45%3A59@%23%24%u5E7F%u53D1%u6052%u751F%u6D88%u8D39%28QDII-ETF%29@%23%24159699; EMFUND3=01-30%2014%3A47%3A24@%23%24%u5357%u65B9%u7EA2%u5229%u4F4E%u6CE250ETF@%23%24515450; EMFUND4=01-31%2015%3A21%3A23@%23%24%u56FD%u8054%u4E2D%u8BC1%u7164%u70AD%u6307%u6570%28LOF%29A@%23%24168204; EMFUND5=02-02%2008%3A34%3A56@%23%24%u666F%u987A%u957F%u57CE%u652F%u67F1%u4EA7%u4E1A%u6DF7%u5408@%23%24260117; EMFUND6=02-02%2010%3A30%3A22@%23%24%u897F%u90E8%u5229%u5F97%u805A%u79BE%u6DF7%u5408A@%23%24007423; EMFUND7=02-26%2016%3A07%3A01@%23%24%u91D1%u9E70%u6DFB%u76C8%u7EAF%u503A%u503A%u5238A@%23%24003384; EMFUND8=02-27%2010%3A51%3A37@%23%24%u534E%u590F%u6210%u957F%u6DF7%u5408@%23%24000001; EMFUND9=02-27 13:53:07@#$%u534E%u6CF0%u67CF%u745E%u5357%u65B9%u4E1C%u82F1%u6052%u751F%u79D1%u6280%28QDII-ETF%29@%23%24513130; st_asi=delete; HAList=ty-0-159001-%u8D27%u5E01ETF%2Cty-0-000001-%u5E73%u5B89%u94F6%u884C%2Cty-0-161127-%u6807%u666E%u751F%u7269%u79D1%u6280LOF%2Cty-1-513130-%u6052%u751F%u79D1%u6280ETF%2Cty-0-300078-%u601D%u521B%u533B%u60E0%2Cty-0-300486-%u4E1C%u6770%u667A%u80FD%2Cty-1-513360-%u6559%u80B2ETF%2Cty-0-300478-%u676D%u5DDE%u9AD8%u65B0%2Cty-0-301589-N%u8BFA%u74E6%2Cty-1-603135-%u4E2D%u91CD%u79D1%u6280; st_pvi=47260449691882; st_sp=2024-01-12%2017%3A28%3A27; st_inirUrl=https%3A%2F%2Fwww.google.com.hk%2F; st_sn=752; st_psi=20240228165720267-112200305283-0910847468
        // Host: fundf10.eastmoney.com
        // Pragma: no-cache
        // Referer: https://fundf10.eastmoney.com/lrfpb_000001.html
        // Sec-Fetch-Dest: script
        // Sec-Fetch-Mode: no-cors
        // Sec-Fetch-Site: same-origin
        // User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36
        // sec-ch-ua: "Not A(Brand";v="99", "Google Chrome";v="121", "Chromium";v="121"
        // sec-ch-ua-mobile: ?0
        // sec-ch-ua-platform: "macOS"

        // Query String Parameters:
        //
        // type: lrfpb
        // code: 000001
        // showtype: 1
        // year: 2021
        // rt: 0.262362624583663

        let request_url = format!(
            "https://fundf10.eastmoney.com/FundArchivesDatas.aspx?type=lrfpb&code={code}&showtype=1&year={year}&rt=0.262362624583663"
        );
        let response = REQWEST_CLIENT.get(request_url)
            .header(reqwest::header::ACCEPT, "*/*")
            .header(reqwest::header::ACCEPT_ENCODING, "gzip, deflate, br")
            .header(reqwest::header::ACCEPT_LANGUAGE, "zh-CN,zh;q=0.9,en;q=0.8")
            .header(reqwest::header::CACHE_CONTROL, "no-cache")
            .header(reqwest::header::CONNECTION, "keep-alive")
            // .header(reqwest::header::COOKIE, "qgqp_b_id=d6299473145338f68687834ba37fa719; st_si=77687174285423; AUTH_FUND.EASTMONEY.COM_GSJZ=AUTH*TTJJ*TOKEN; EMFUND0=01-16%2016%3A57%3A06@%23%24%u4E2D%u822A%u65B0%u8D77%u822A%u7075%u6D3B%u914D%u7F6E%u6DF7%u5408A@%23%24005537; EMFUND1=01-16%2017%3A04%3A53@%23%24%u91D1%u4FE1%u6D88%u8D39%u5347%u7EA7%u80A1%u7968A@%23%24006692; EMFUND2=01-16%2020%3A01%3A56@%23%24%u5357%u65B9%u4E0A%u8BC1380ETF@%23%24510290; EMFUND3=01-29%2015%3A02%3A07@%23%24%u534E%u590F%u4E2D%u8BC1%u5168%u6307%u623F%u5730%u4EA7ETF@%23%24515060; EMFUND4=01-30%2014%3A41%3A22@%23%24%u534E%u6CF0%u67CF%u745E%u6D88%u8D39%u6210%u957F%u6DF7%u5408@%23%24001069; EMFUND5=01-30%2014%3A45%3A59@%23%24%u5E7F%u53D1%u6052%u751F%u6D88%u8D39%28QDII-ETF%29@%23%24159699; EMFUND6=01-30%2014%3A47%3A24@%23%24%u5357%u65B9%u7EA2%u5229%u4F4E%u6CE250ETF@%23%24515450; EMFUND7=01-31%2015%3A21%3A23@%23%24%u56FD%u8054%u4E2D%u8BC1%u7164%u70AD%u6307%u6570%28LOF%29A@%23%24168204; EMFUND8=02-02%2008%3A34%3A56@%23%24%u666F%u987A%u957F%u57CE%u652F%u67F1%u4EA7%u4E1A%u6DF7%u5408@%23%24260117; EMFUND9=02-02 10:30:22@#$%u897F%u90E8%u5229%u5F97%u805A%u79BE%u6DF7%u5408A@%23%24007423; st_asi=delete; HAList=ty-1-513130-%u6052%u751F%u79D1%u6280ETF%2Cty-0-300078-%u601D%u521B%u533B%u60E0%2Cty-0-300486-%u4E1C%u6770%u667A%u80FD%2Cty-1-513360-%u6559%u80B2ETF%2Cty-0-300478-%u676D%u5DDE%u9AD8%u65B0%2Cty-0-301589-N%u8BFA%u74E6%2Cty-1-603135-%u4E2D%u91CD%u79D1%u6280%2Cty-0-000001-%u5E73%u5B89%u94F6%u884C%2Cty-0-002400-%u7701%u5E7F%u96C6%u56E2%2Cty-1-601127-%u8D5B%u529B%u65AF; st_pvi=47260449691882; st_sp=2024-01-12%2017%3A28%3A27; st_inirUrl=https%3A%2F%2Fwww.google.com.hk%2F; st_sn=438; st_psi=20240226102321941-113200301321-9024721593")
            .header(reqwest::header::HOST, "fundf10.eastmoney.com")
            .header(reqwest::header::PRAGMA, "no-cache")
            .header(
                reqwest::header::REFERER,
                &format!("https://fundf10.eastmoney.com/lrfpb_{code}.html")
            )
            .header(
                reqwest::header::USER_AGENT,
                "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36"
            )
            .header(
                "Sec-Ch-Ua",
                "\"Not A(Brand\";v=\"99\", \"Google Chrome\";v=\"121\", \"Chromium\";v=\"121\""
            )
            .header("Sec-Ch-Ua-Mobile", "?0")
            .header("Sec-Ch-Ua-Platform", "\"macOS\"")
            .header("Sec-Fetch-Dest", "empty")
            .header("Sec-Fetch-Mode", "cors")
            .header("Sec-Fetch-Site", "same-origin")
            .send().await?;

        let body = response.text().await?;
        // println!("*** body={}\n\n", body);

        let json_body = body
            .replace("var apidata=", "")
            .replace(";", "")
            .replace("content:", "\"content\":")
            .replace("arryear:", "\"arryear\":")
            .replace("curyear:", "\"curyear\":");
        // println!("\n\njson_body: {:?}", json_body);

        let api_data: ApiData = serde_json::from_str(&json_body)?;

        println!("\n\napi_data= {:?}", api_data);

        Self::extract_data(&api_data.content)
    }

    fn extract_data(html: &str) -> Result<Vec<FundLrfpb>> {
        let mut fund_lrfpbs = Vec::new();
        let document = Html::parse_document(html);
        // <table class="w782 comm lrb">
        let table_th_selector = Selector::parse("table.w782.comm.lrb thead tr th").map_err(|e|
            anyhow!("{}", e)
        )?;

        for (index, th) in document.select(&table_th_selector).enumerate() {
            println!("\n\nth={:?}", th.text().collect::<String>());
            if index > 0 {
                let mut fund_lrfpb = FundLrfpb {
                    ..Default::default()
                };
                fund_lrfpb.report_date = th.text().collect::<String>();
                fund_lrfpbs.push(fund_lrfpb);
            }
        }

        let table_bodytr_selector = Selector::parse("table.w782.comm.lrb tbody tr").map_err(|e|
            anyhow!("{}", e)
        )?;
        let td_selector = &Selector::parse("td").map_err(|e| anyhow!("{}", e))?;

        for (_tri, tr) in document.select(&table_bodytr_selector).enumerate() {
            let mut items = vec![
                FundLrfpbData {
                    ..Default::default()
                },
                FundLrfpbData {
                    ..Default::default()
                },
                FundLrfpbData {
                    ..Default::default()
                },
                FundLrfpbData {
                    ..Default::default()
                }
            ];
            for (tdi, td) in tr.select(td_selector).enumerate() {
                if tdi == 0 {
                    let title = td.text().collect::<String>();
                    items.get_mut(0).unwrap().item_title = title.clone();
                    items.get_mut(1).unwrap().item_title = title.clone();
                    items.get_mut(2).unwrap().item_title = title.clone();
                    items.get_mut(3).unwrap().item_title = title;
                } else if tdi < 5 {
                    items.get_mut(tdi - 1).unwrap().value = td.text().collect::<String>();

                    fund_lrfpbs
                        .get_mut(tdi - 1)
                        .unwrap()
                        .data.push(
                            items
                                .get(tdi - 1)
                                .unwrap()
                                .clone()
                        );
                }
            }
        }

        Ok(fund_lrfpbs)
    }
}

#[cfg(test)]
mod test {
    use crate::fund::fund_lrfpb::FundLrfpb;

    #[tokio::test]
    async fn test_fund_lrfpbs_2021() {
        let result = FundLrfpb::fund_lrfpbs("000001", "2021").await;

        // println!("result={:?}", result);

        assert_eq!(result.is_ok(), true);
        if let Ok(fund_lrfpbs) = result {
            assert_eq!(fund_lrfpbs.len(), 4);
            for (index, fund_lrfpb) in fund_lrfpbs.iter().enumerate() {
                println!("fund_lrfpb[{index}]= {:?}\n", fund_lrfpb);
            }
        }
    }

    #[tokio::test]
    async fn test_fund_lrfpbs() {
        let result = FundLrfpb::fund_lrfpbs("000001", "").await;

        // println!("result={:?}", result);

        assert_eq!(result.is_ok(), true);
        if let Ok(fund_lrfpbs) = result {
            for (index, fund_lrfpb) in fund_lrfpbs.iter().enumerate() {
                println!("fund_lrfpb[{index}]= {:?}\n", fund_lrfpb);
            }
        }
    }
}
